

package com.hazelcast.test.mocknetwork;

import com.hazelcast.cluster.Address;
import com.hazelcast.instance.AddressPicker;
import com.hazelcast.instance.EndpointQualifier;

import java.nio.channels.ServerSocketChannel;
import java.util.HashMap;
import java.util.Map;

import static com.hazelcast.instance.EndpointQualifier.MEMBER;

class StaticAddressPicker implements AddressPicker {

    private final Address thisAddress;

    StaticAddressPicker(Address thisAddress) {
        this.thisAddress = thisAddress;
    }

    @Override
    public void pickAddress() {
    }

    @Override
    public Address getBindAddress(EndpointQualifier qualifier) {
        return thisAddress;
    }

    @Override
    public Address getPublicAddress(EndpointQualifier qualifier) {
        return thisAddress;
    }

    @Override
    public Map<EndpointQualifier, Address> getPublicAddressMap() {
        HashMap<EndpointQualifier, Address> publicAddressMap = new HashMap<>();
        publicAddressMap.put(MEMBER, thisAddress);
        return publicAddressMap;
    }

    @Override
    public Map<EndpointQualifier, Address> getBindAddressMap() {
        HashMap<EndpointQualifier, Address> bindAddressMap = new HashMap<>();
        bindAddressMap.put(MEMBER, thisAddress);
        return bindAddressMap;
    }

    @Override
    public ServerSocketChannel getServerSocketChannel(EndpointQualifier qualifier) {
        return null;
    }

    @Override
    public Map<EndpointQualifier, ServerSocketChannel> getServerSocketChannels() {
        return null;
    }

}
